import collections
import math
import torch
from torch import nn
from d2l import torch as d2l
from s03_create_vocab import load_data_nmt
import matplotlib.pyplot as plt
from s04_encoder_and_decoder import Encoder, Decoder, EncoderDecoder

train_iter, src_vocab, tgt_vocab = load_data_nmt(32, 30)
print(len(train_iter))
print(len(src_vocab))
print(len(tgt_vocab))
# print(src_vocab)
# print(tgt_vocab)
